#!/bin/bash
###
 # @Version: 2.0
 # @Author: Yue Zhong
 # @Date: 2024-09-26 20:02:26
 # @Description: 
 # @LastEditors: Yue Zhong
 # @LastEditTime: 2024-10-29 16:37:27
### 

gpu=0
dataset_dir='./data'
dataset_name=SVHN

prefix_resume='./macro_search_result/uniform_sampling/AutoSNN_16_'
suffix_resume='_SNN_Adam_600ep_2022/checkpoint.pth.tar'
resume=$prefix_resume$dataset_name$suffix_resume

python train_supernet/train.py \
    --gpu $gpu \
    --T 8 --init_tau 2.0 --v_threshold 1.0 --neuron PLIF \
    --epochs 600 \
    --dataset_dir $dataset_dir \
    --dataset_name $dataset_name \
    --save uniform_sampling \
    --search_space AutoSNN_16 \
    --seed 2022 \
    --resume $resume

